"""
!/usr/bin/env python
-*- coding: utf-8 -*-
@Author  : MarkJia
@Site    : 
@File    : main_page.py
"""
import time

"""
首页页面对象
"""

# 导包时需要局部导入 。 有 return 时，按住 ctrl + 方法，会自动跳转到对应方法


from selenium.webdriver.common.by import By
from webtest.autowebui.page_object.base_page import BasePage
from webtest.autowebui.utils.Logger import logger


class Mainpage(BasePage):
    # 设定首页的base_url
    _base_url = 'https://work.weixin.qq.com/wework_admin/frame#index'

    # 提取添加成员元素
    _add_loc = (By.CSS_SELECTOR, ".ww_indexImg_AddMember")

    # 通讯录元素
    _tongxunlu_loc = (By.XPATH, "//span[text()='通讯录']")

    # 一旦涉及页面跳转的方法，可以直接 return 要跳转页面的实例对象。形成链式调用
    def goto_contact(self):
        from webtest.autowebui.page_object.contact_page import ContactPage
        """
        跳转通讯录页面的方法
        :return:
        """
        self.find_ele(self._tongxunlu_loc).click()
        time.sleep(5)
        return ContactPage(self.driver)

    def goto_add_memeber(self):
        from webtest.autowebui.page_object.add_member_page import AddmemberPage
        """
        跳转到添加成员页面的方法
        :return:
        """
        self.find_ele(self._add_loc).click()
        logger.info('首页点击添加成员按钮')
        return AddmemberPage(self.driver)
